--- ../../plenoxels/opt/util/llff_dataset.py	2024-02-13 12:06:07.930166019 -0500
+++ ../google/plenoxels/opt/util/llff_dataset.py	2024-02-13 10:46:24.456645151 -0500
@@ -52,6 +52,7 @@
         render_style="",
         hold_every=8,
         offset=250,
+        n_images=-1,
         **kwargs
     ):
         super().__init__()
@@ -90,6 +91,10 @@
         self.is_train_split = is_train_split
 
         self._load_images()
+        if n_images != -1:
+            self.gt = self.gt[:n_images]
+            self.c2w = self.c2w[:n_images]
+            
         self.n_images, self.h_full, self.w_full, _ = self.gt.shape
         assert self.h_full == self.sfm.ref_cam["height"]
         assert self.w_full == self.sfm.ref_cam["width"]
